# config.py
import os

# 从环境变量获取API密钥，如果不存在则使用默认值
DASHSCOPE_API_KEY = os.getenv('DASHSCOPE_API_KEY', 'sk-4458a4ebf1c146f1b59434c22aa85aea')

# 雷达仿真系统配置
RADAR_LOG_FILE = "logs/radar_system.log"
CHAT_LOG_FILE = "logs/gui_chat.log"

# LLM模型配置
LLM_MODEL = "qwen-plus"
LLM_TEMPERATURE = 0.3

# 系统提示词
RADAR_SYSTEM_PROMPT = """你是一个雷达仿真系统的智能助手。你可以帮助用户：

1. 管理雷达目标（添加、删除目标）
2. 设置雷达参数（功率、频率、增益等）
3. 控制仿真（开始、停止、步进）
4. 查询系统状态
5. 回答关于雷达仿真的问题

请用友好、专业的语调回复用户。如果用户想要执行具体操作，请给出清晰的指令说明。"""

def setup_environment():
    """设置环境变量和配置"""
    # 设置DashScope API密钥
    os.environ['DASHSCOPE_API_KEY'] = DASHSCOPE_API_KEY

    # 创建必要的目录
    os.makedirs("logs", exist_ok=True)
    os.makedirs("data/export", exist_ok=True)

    print(f"SUCCESS: API密钥已配置")
    print(f"SUCCESS: 环境设置完成")